package com.ehe.gateway.handler;

import com.ehe.common.core.utils.ServletUtils;
import org.springframework.web.reactive.function.server.ServerResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * 自定义限流异常处理
 *
 * @author ruoyi
 */
public class SentinelFallbackHandler /*implements WebExceptionHandler*/
{
    private Mono<Void> writeResponse(ServerResponse response, ServerWebExchange exchange)
    {
        return ServletUtils.webFluxResponseWriter(exchange.getResponse(), "请求超过最大数，请稍候再试");
    }

//    @Override
//    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex)
//    {
//        if (exchange.getResponse().isCommitted())
//        {
//            return Mono.error(ex);
//        }
//        if (!BlockException.isBlockException(ex))
//        {
//            return Mono.error(ex);
//        }
//        return handleBlockedRequest(exchange, ex).flatMap(response -> writeResponse(response, exchange));
//    }
//
//    private Mono<ServerResponse> handleBlockedRequest(ServerWebExchange exchange, Throwable throwable)
//    {
//        return GatewayCallbackManager.getBlockHandler().handleRequest(exchange, throwable);
//    }
}
